<?php

// +----------------------------------------------------------------------
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
// +----------------------------------------------------------------------
// | Copyright (c) 2009 http://thinkphp.cn All rights reserved.
// +----------------------------------------------------------------------
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
// +----------------------------------------------------------------------
// | Author: liu21st <liu21st@gmail.com>
// +----------------------------------------------------------------------
// $Id: UploadFile_sae.class.php 1271 2012-12-02 12:17:00Z luofei614@126.com $

/**
 +------------------------------------------------------------------------------
 * 文件上传类
 +------------------------------------------------------------------------------
 * @category   ORG
 * @package  ORG
 * @subpackage  Net
 * @author    liu21st <liu21st@gmail.com>
 * @version   $Id: UploadFile_sae.class.php 1271 2012-12-02 12:17:00Z luofei614@126.com $
 +------------------------------------------------------------------------------
 */
class UploadFile {//类定义开始
	// 上传文件的最大值

	private $config =   array(
        'maxSize'           =>  -1,    // 上传文件的最大值
        'supportMulti'      =>  true,    // 是否支持多文件上传
        'allowExts'         =>  array(),    // 允许上传的文件后缀 留空不作后缀检查
        'allowTypes'        =>  array(),    // 允许上传的文件类型 留空不做检查
        'thumb'             =>  false,    // 使用对上传图片进行缩略图处理
        'imageClassPath'    =>  'ORG.Util.Image',    // 图库类包路径
        'thumbMaxWidth'     =>  '',// 缩略图最大宽度
        'thumbMaxHeight'    =>  '',// 缩略图最大高度
        'thumbPrefix'       =>  'thumb_',// 缩略图前缀
        'thumbSuffix'       =>  '',
        'thumbPath'         =>  '',// 缩略图保存路径
        'thumbFile'         =>  '',// 缩略图文件名
        'thumbExt'          =>  '',// 缩略图扩展名        
        'thumbRemoveOrigin' =>  false,// 是否移除原图
        'zipImages'         =>  false,// 压缩图片文件上传
        'autoSub'           =>  false,// 启用子目录保存文件
        'subType'           =>  'hash',// 子目录创建方式 可以使用hash date
        'dateFormat'        =>  'Ymd',
        'hashLevel'         =>  1, // hash的目录层次
        'savePath'          =>  '',// 上传文件保存路径
        'autoCheck'         =>  true, // 是否自动检查附件
        'uploadReplace'     =>  false,// 存在同名是否覆盖
        'saveRule'          =>  'uniqid',// 上传文件命名规则
        'hashType'          =>  'md5_file',// 上传文件Hash规则函数名
	//sae平台下特有属性
        'expires'=>'',
        'encoding'=>'',
        'type'=>'',
        'private'=>'',
        'compress'=>false
	);

	// 错误信息
	private $error = '';
	// 上传成功的文件信息
	private $uploadFileInfo ;
	//[sae] storage的domain
	private $domain;
	public function __get($name){
		if(isset($this->config[$name])) {
			return $this->config[$name];
		}
		return null;
	}

	public function __set($name,$value){
		if(isset($this->config[$name])) {
			$this->config[$name]    =   $value;
		}
	}
	public function __isset($name){
		return isset($this->config[$name]);
	}

	/**
	 +----------------------------------------------------------
	 * 架构函数
	 +----------------------------------------------------------
	 * @access public
	 +----------------------------------------------------------
	 */
	public function __construct($config=array()) {
		if(is_array($config)) {
			$this->config   =   array_merge($this->config,$config);
		}
	}

	/**
	 +----------------------------------------------------------
	 * 上传一个文件
	 +----------------------------------------------------------
	 * @access public
	 +----------------------------------------------------------
	 * @param mixed $name 数据
	 * @param string $value  数据表名
	 +----------------------------------------------------------
	 * @return string
	 +----------------------------------------------------------
	 * @throws ThinkExecption
	 +----------------------------------------------------------
	 */
	private function save($file) {
		$filename = $file['savepath'] . $file['savename'];
		$s = Think::instance('SaeStorage');
		if (!$this->uploadReplace && $s->fileExists($this->domain, $filename)) {
			// 不覆盖同名文件
			$this->error = '文件已经存在！' . $filename;
			return false;
		}
		// 如果是图像文件 检测文件格式
		if (in_array(strtolower($file['extension']), array('gif', 'jpg', 'jpeg', 'bmp', 'png', 'swf')) && false === getimagesize($file['tmp_name'])) {
			$this->error = '非法图像文件';
			return false;
		}
		$attr=array();
		$attrs=array('expires','encoding','type','private');
		foreach ($attrs as $key => $value)
		if(!empty($this->config[$key])) $attr[$key]=$value;
		if($this->compress) $attr['encoding']='gzip';
		//[sae] 上传文件
		if (!$this->thumbRemoveOrigin && !$s->upload($this->domain, $filename, $file['tmp_name'],$attr,$compress) ) {
			$this->error = '文件上传失败'.$s->errmsg();
			return false;
		}
		if ($this->thumb && in_array(strtolower($file['extension']), array('gif', 'jpg', 'jpeg', 'bmp', 'png'))) {
			$image = getimagesize($file['tmp_name']);
			if (false !== $image) {
				//是图像文件生成缩略图
				$thumbWidth = explode(',', $this->thumbMaxWidth);
				$thumbHeight = explode(',', $this->thumbMaxHeight);
				$thumbPrefix = explode(',', $this->thumbPrefix);
				$thumbSuffix = explode(',', $this->thumbSuffix);
				$thumbFile = explode(',', $this->thumbFile);
				$thumbPath    =  $this->thumbPath?$this->thumbPath:dirname($filename).'/';
				$thumbExt       =   $this->thumbExt ? $this->thumbExt : $file['extension']; //自定义缩略图扩展名
				//[sae] 定义缩略图目录时，判断doamin
				$domain = $this->thumbPath ? $this->thumbDomain : $this->domain;
				//[sae] 用自带image类生成缩略图
				// $realFilename = $this->autoSub ? basename($file['savename']) : $file['savename'];
				$srcWidth = $image[0];
				$srcHeight = $image[1];
				$img = Think::instance('SaeImage');
				for ($i = 0, $len = count($thumbWidth); $i < $len; $i++) {
					$scale = min($thumbWidth[$i] / $srcWidth, $thumbHeight[$i] / $srcHeight); // 计算缩放比例
					if ($scale >= 1) {
						// 超过原图大小不再缩略
						$width = $srcWidth;
						$height = $srcHeight;
					} else {
						// 缩略图尺寸
						$width = (int) ($srcWidth * $scale);
						$height = (int) ($srcHeight * $scale);
					}

					if(!empty($thumbFile[$i])) {
						$thumbname  =   $thumbFile[$i];
					}else{
						$prefix =   isset($thumbPrefix[$i])?$thumbPrefix[$i]:$thumbPrefix[0];
						$suffix =   isset($thumbSuffix[$i])?$thumbSuffix[$i]:$thumbSuffix[0];
						$thumbname  =   $prefix.basename($filename,'.'.$thumbExt).$suffix;
					}

					$img->setData(file_get_contents($file['tmp_name']));
					$img->resize($width, $height);
					$new_data = $img->exec();
					if (!$s->write($domain, $thumbPath . $thumbname.'.'.$thumbExt, $new_data)) {
						$this->error = '生成缩略图失败！'.$this->errmsg();
						return false;
					}
				}
			}
		}
		if ($this->zipImags) {
			// TODO 对图片压缩包在线解压
		}
		return true;
	}

	//[sae]获得domain，改变path
	private function getDomain($filePath) {
		$arr = explode('/', ltrim($filePath, './'));
		$domain = array_shift($arr);
		$filePath = implode('/', $arr);
		return array($domain, $filePath);
	}

	/**
	 +----------------------------------------------------------
	 * 上传所有文件
	 +----------------------------------------------------------
	 * @access public
	 +----------------------------------------------------------
	 * @param string $savePath  上传文件保存路径
	 +----------------------------------------------------------
	 * @return string
	 +----------------------------------------------------------
	 * @throws ThinkExecption
	 +----------------------------------------------------------
	 */
	public function upload($savePath ='') {
		//如果不指定保存文件名，则由系统默认
		if (empty($savePath))
		$savePath = $this->savePath;
		//[sae] 去掉检查上传目录
		$fileInfo = array();
		$isUpload = false;
		//[sae] 分析出domain，第一个目录为domain
		list($this->domain, $savePath) = $this->getDomain($savePath);
		//[sae] 分析缩略图保存地址
		if ($this->thumb && $this->thumbPath)
		list($this->thumbDomain, $this->thumbPath) = $this->getDomain($this->thumbPath);
		// 获取上传的文件信息
		// 对$_FILES数组信息处理
		$files = $this->dealFiles($_FILES);
		foreach ($files as $key => $file) {
			//过滤无效的上传
			if (!empty($file['name'])) {
				//登记上传文件的扩展信息
				if(!isset($file['key']))   $file['key']    =   $key;
				$file['extension'] = $this->getExt($file['name']);
				$file['savepath'] = $savePath;
				$file['savename'] = $this->getSaveName($file);

				// 自动检查附件
				if ($this->autoCheck) {
					if (!$this->check($file))
					return false;
				}
				//保存上传文件
				if (!$this->save($file))
				return false;
				if (function_exists($this->hashType)) {
					$fun = $this->hashType;
					$file['hash'] = $fun($this->autoCharset($file['tmp_name'], 'utf-8', 'gbk'));
				}
				//上传成功后保存文件信息，供其他地方调用
				unset($file['tmp_name'], $file['error']);
				$fileInfo[] = $file;
				$isUpload = true;
			}
		}
		if ($isUpload) {
			$this->uploadFileInfo = $fileInfo;
			return true;
		} else {
			$this->error = '没有选择上传文件';
			return false;
		}
	}

	/**
	 +----------------------------------------------------------
	 * 上传单个上传字段中的文件 支持多附件
	 +----------------------------------------------------------
	 * @access public
	 +----------------------------------------------------------
	 * @param array $file  上传文件信息
	 * @param string $savePath  上传文件保存路径
	 +----------------------------------------------------------
	 * @return string
	 +----------------------------------------------------------
	 * @throws ThinkExecption
	 +----------------------------------------------------------
	 */
	public function uploadOne($file, $savePath='') {
		//如果不指定保存文件名，则由系统默认
		if (empty($savePath))
		$savePath = $this->savePath;
		//[sae] 不检查上传目录
		//[sae] 分析出domain，第一个目录为domain
		list($this->domain, $savePath) = $this->getDomain($savePath);
		//[sae] 分析缩略图保存地址
		if ($this->thumb && $this->thumbPath)
		list($this->thumbDomain, $this->thumbPath) = $this->getDomain($this->thumbPath);
		//过滤无效的上传
		if (!empty($file['name'])) {
			$fileArray = array();
			if (is_array($file['name'])) {
				$keys = array_keys($file);
				$count = count($file['name']);
				for ($i = 0; $i < $count; $i++) {
					foreach ($keys as $key)
					$fileArray[$i][$key] = $file[$key][$i];
				}
			} else {
				$fileArray[] = $file;
			}
			$info = array();
			foreach ($fileArray as $key => $file) {
				//登记上传文件的扩展信息
				$file['extension'] = $this->getExt($file['name']);
				$file['savepath'] = $savePath;
				$file['savename'] = $this->getSaveName($file);
				// 自动检查附件
				if ($this->autoCheck) {
					if (!$this->check($file))
					return false;
				}
				//保存上传文件
				if (!$this->save($file))
				return false;
				if (function_exists($this->hashType)) {
					$fun = $this->hashType;
					$file['hash'] = $fun($this->autoCharset($file['savepath'] . $file['savename'], 'utf-8', 'gbk'));
				}
				unset($file['tmp_name'], $file['error']);
				$info[] = $file;
			}
			// 返回上传的文件信息
			return $info;
		} else {
			$this->error = '没有选择上传文件';
			return false;
		}
	}

	/**
	 +----------------------------------------------------------
	 * 转换上传文件数组变量为正确的方式
	 +----------------------------------------------------------
	 * @access private
	 +----------------------------------------------------------
	 * @param array $files  上传的文件变量
	 +----------------------------------------------------------
	 * @return array
	 +----------------------------------------------------------
	 */
	private function dealFiles($files) {
		$fileArray  = array();
		$n          = 0;
		foreach ($files as $key=>$file){
			if(is_array($file['name'])) {
				$keys       =   array_keys($file);
				$count      =   count($file['name']);
				for ($i=0; $i<$count; $i++) {
					$fileArray[$n]['key'] = $key;
					foreach ($keys as $_key){
						$fileArray[$n][$_key] = $file[$_key][$i];
					}
					$n++;
				}
			}else{
				$fileArray[$key] = $file;
			}
		}
		return $fileArray;
	}

	/**
	 +----------------------------------------------------------
	 * 获取错误代码信息
	 +----------------------------------------------------------
	 * @access public
	 +----------------------------------------------------------
	 * @param string $errorNo  错误号码
	 +----------------------------------------------------------
	 * @return void
	 +----------------------------------------------------------
	 * @throws ThinkExecption
	 +----------------------------------------------------------
	 */
	protected function error($errorNo) {
		switch ($errorNo) {
			case 1:
				$this->error = '上传的文件超过了 php.ini 中 upload_max_filesize 选项限制的值';
				break;
			case 2:
				$this->error = '上传文件的大小超过了 HTML 表单中 MAX_FILE_SIZE 选项指定的值';
				break;
			case 3:
				$this->error = '文件只有部分被上传';
				break;
			case 4:
				$this->error = '没有文件被上传';
				break;
			case 6:
				$this->error = '找不到临时文件夹';
				break;
			case 7:
				$this->error = '文件写入失败';
				break;
			default:
				$this->error = '未知上传错误！';
		}
		return;
	}

	/**
	 +----------------------------------------------------------
	 * 根据上传文件命名规则取得保存文件名
	 +----------------------------------------------------------
	 * @access private
	 +----------------------------------------------------------
	 * @param string $filename 数据
	 +----------------------------------------------------------
	 * @return string
	 +----------------------------------------------------------
	 */
	private function getSaveName($filename) {
		$rule = $this->saveRule;
		if (empty($rule)) {//没有定义命名规则，则保持文件名不变
			$saveName = $filename['name'];
		} else {
			if (function_exists($rule)) {
				//使用函数生成一个唯一文件标识号
				$saveName = $rule() . "." . $filename['extension'];
			} else {
				//使用给定的文件名作为标识号
				$saveName = $rule . "." . $filename['extension'];
			}
		}
		if ($this->autoSub) {
			// 使用子目录保存文件
			$filename['savename'] = $saveName;
			$saveName = $this->getSubName($filename) . $saveName;
		}
		return $saveName;
	}

	/**
	 +----------------------------------------------------------
	 * 获取子目录的名称
	 +----------------------------------------------------------
	 * @access private
	 +----------------------------------------------------------
	 * @param array $file  上传的文件信息
	 +----------------------------------------------------------
	 * @return string
	 +----------------------------------------------------------
	 */
	private function getSubName($file) {
		switch ($this->subType) {
			case 'date':
				$dir = date($this->dateFormat, time()).'/';
				break;
			case 'hash':
			default:
				$name = md5($file['savename']);
				$dir = '';
				for ($i = 0; $i < $this->hashLevel; $i++) {
					$dir .= $name{$i} . '/';
				}
				break;
		}
		//[sae] 不用建立建立子目录
		// if (!is_dir($file['savepath'] . $dir)) {
		//     mkdir($file['savepath'] . $dir,0777,true);
		// }
		return $dir;
	}

	/**
	 +----------------------------------------------------------
	 * 检查上传的文件
	 +----------------------------------------------------------
	 * @access private
	 +----------------------------------------------------------
	 * @param array $file 文件信息
	 +----------------------------------------------------------
	 * @return boolean
	 +----------------------------------------------------------
	 */
	private function check($file) {
		if ($file['error'] !== 0) {
			//文件上传失败
			//捕获错误代码
			$this->error($file['error']);
			return false;
		}
		//文件上传成功，进行自定义规则检查
		//检查文件大小
		if (!$this->checkSize($file['size'])) {
			$this->error = '上传文件大小不符！';
			return false;
		}

		//检查文件Mime类型
		if (!$this->checkType($file['type'])) {
			$this->error = '上传文件MIME类型不允许！';
			return false;
		}
		//检查文件类型
		if (!$this->checkExt($file['extension'])) {
			$this->error = '上传文件类型不允许';
			return false;
		}

		//检查是否合法上传
		if (!$this->checkUpload($file['tmp_name'])) {
			$this->error = '非法上传文件！';
			return false;
		}
		return true;
	}

	// 自动转换字符集 支持数组转换
	private function autoCharset($fContents, $from='gbk', $to='utf-8') {
		$from = strtoupper($from) == 'UTF8' ? 'utf-8' : $from;
		$to = strtoupper($to) == 'UTF8' ? 'utf-8' : $to;
		if (strtoupper($from) === strtoupper($to) || empty($fContents) || (is_scalar($fContents) && !is_string($fContents))) {
			//如果编码相同或者非字符串标量则不转换
			return $fContents;
		}
		if (function_exists('mb_convert_encoding')) {
			return mb_convert_encoding($fContents, $to, $from);
		} elseif (function_exists('iconv')) {
			return iconv($from, $to, $fContents);
		} else {
			return $fContents;
		}
	}

	/**
	 +----------------------------------------------------------
	 * 检查上传的文件类型是否合法
	 +----------------------------------------------------------
	 * @access private
	 +----------------------------------------------------------
	 * @param string $type 数据
	 +----------------------------------------------------------
	 * @return boolean
	 +----------------------------------------------------------
	 */
	private function checkType($type) {
		if (!empty($this->allowTypes))
		return in_array(strtolower($type), $this->allowTypes);
		return true;
	}

	/**
	 +----------------------------------------------------------
	 * 检查上传的文件后缀是否合法
	 +----------------------------------------------------------
	 * @access private
	 +----------------------------------------------------------
	 * @param string $ext 后缀名
	 +----------------------------------------------------------
	 * @return boolean
	 +----------------------------------------------------------
	 */
	private function checkExt($ext) {
		if (!empty($this->allowExts))
		return in_array(strtolower($ext), $this->allowExts, true);
		return true;
	}

	/**
	 +----------------------------------------------------------
	 * 检查文件大小是否合法
	 +----------------------------------------------------------
	 * @access private
	 +----------------------------------------------------------
	 * @param integer $size 数据
	 +----------------------------------------------------------
	 * @return boolean
	 +----------------------------------------------------------
	 */
	private function checkSize($size) {
		return!($size > $this->maxSize) || (-1 == $this->maxSize);
	}

	/**
	 +----------------------------------------------------------
	 * 检查文件是否非法提交
	 +----------------------------------------------------------
	 * @access private
	 +----------------------------------------------------------
	 * @param string $filename 文件名
	 +----------------------------------------------------------
	 * @return boolean
	 +----------------------------------------------------------
	 */
	private function checkUpload($filename) {
		return is_uploaded_file($filename);
	}

	/**
	 +----------------------------------------------------------
	 * 取得上传文件的后缀
	 +----------------------------------------------------------
	 * @access private
	 +----------------------------------------------------------
	 * @param string $filename 文件名
	 +----------------------------------------------------------
	 * @return boolean
	 +----------------------------------------------------------
	 */
	private function getExt($filename) {
		$pathinfo = pathinfo($filename);
		return $pathinfo['extension'];
	}

	/**
	 +----------------------------------------------------------
	 * 取得上传文件的信息
	 +----------------------------------------------------------
	 * @access public
	 +----------------------------------------------------------
	 * @return array
	 +----------------------------------------------------------
	 */
	public function getUploadFileInfo() {
		return $this->uploadFileInfo;
	}

	/**
	 +----------------------------------------------------------
	 * 取得最后一次错误信息
	 +----------------------------------------------------------
	 * @access public
	 +----------------------------------------------------------
	 * @return string
	 +----------------------------------------------------------
	 */
	public function getErrorMsg() {
		return $this->error;
	}

}